package practise;

import java.util.PriorityQueue;

public class SmallestK {
    public int[] smallestK(int[] arr, int k) {
        int[] ret = new int[k];

        if(arr.length == 0 || k == 0) {
            return ret;
        }
        // 采用 lambda表达式写比较器 创建大根堆
        PriorityQueue<Integer> queue = new PriorityQueue<>((o1, o2) -> -o1.compareTo(o2));
        for(int i = 0; i < k; i++) {
            queue.offer(arr[i]);
        }

        for(int i = k; i < arr.length; i++) {
            if(queue.peek() > arr[i]) {
                // 比当前堆中最大要小
                queue.poll();
                queue.offer(arr[i]);
            }
        }

        for(int i = 0; i < k; i++) {
            ret[i] = queue.poll();
        }

        return ret;
    }
}
